"""
需求：定义一个登录函数，准备参数接受测试数据（成功数据、失败失败）
     返回函数处理结果
"""

# 导包
import requests
import config


# 定义获取验证码函数
def get_verify_code():
    return requests.get(url=f"{config.BASE_URL}/api/captchaImage")


# 定义登录函数
def login(username, password, uuid, code="2"):
    url = f"{config.BASE_URL}/api/login"
    test_data = {
        "username": username,
        "password": password,
        "code": code,
        "uuid": uuid
    }
    return requests.post(url=url, json=test_data)


if __name__ == '__main__':
    # 获取验证码中uuid
    uuid = get_verify_code().json().get("uuid")
    # 登录成功
    response = login(config.USERNAME, config.PASSWORD, uuid)
    # 提取响应数据
    print(response.json())
    print(response.json().get("token"))
